/*
 * Copyright 2011 The Rusted Roof, LLC.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package org.proteusframework.platformservice.persistence.c14n;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Tacoma Four
 */
final class ShortArrayC14NHandler extends AbstractC14NHandler
{
    @Override
    public Object fromC14N(byte[] bytes)
    {
        int idx = 0;

        List<Short> shortList = new ArrayList<Short>();
        DataInputStream shortDis = new DataInputStream(new ByteArrayInputStream(bytes));

        try
        {
            while (true)
            {
                shortList.add(shortDis.readShort());
            }
        } catch (EOFException ignore)
        {

        } catch (IOException e)
        {
            e.printStackTrace();
        }

        short[] shortValues = new short[shortList.size()];
        for (Short curValue : shortList)
        {
            shortValues[idx] = curValue;
            idx++;
        }


        return shortValues;
    }

    @Override
    public byte[] toC14N(Object object)
    {
        byte[] shortBytes = new byte[0];

        short[] shortValues = (short[]) object;
        ByteArrayOutputStream shortBaos = new ByteArrayOutputStream();
        DataOutputStream shortDos = new DataOutputStream(shortBaos);

        String shortHeader = PACKED_BLOB_PREFIX + transposeClassName(short[].class.getName()) + PACKED_BLOB_SUFFIX;
        byte[] shortHeaderBytes = shortHeader.getBytes();
        try
        {
            shortDos.write(shortHeaderBytes, 0, shortHeaderBytes.length);
            for (short curValue : shortValues)
            {
                shortDos.writeShort(curValue);
            }

            shortDos.flush();
            shortDos.close();

            shortBytes = shortBaos.toByteArray();
        } catch (IOException e)
        {
            e.printStackTrace();
        }

        return shortBytes;
    }
}
